NUM_CLASSES = 8

BATCH_SIZE = 16 

NUM_EPOCHS = 25

##训练完成，权重文件的保存路径,默认保存在trained_models下
TRAINED_WEIGHTS = './trained_models/resnet.pth'

#数据集的存放位置
TRAIN_DATASET_DIR = '.data/train'
VALID_DATASET_DIR = '.data/val'


